package com.closer.server.mapper.problem;

import com.mybatisflex.core.BaseMapper;
import com.closer.server.biz.problem.entity.TestCase;
import com.mybatisflex.core.query.QueryWrapper;

import java.util.List;

/**
 * 测试样例表 映射层。
 *
 * @author closer
 */
public interface TestCaseMapper extends BaseMapper<TestCase> {

    /**
     * 根据题目id查询测试样例 id
     * @param problemId 题目id
     * @return 测试样例 id
     */
    default Long getByProblemId(Long problemId) {
        var queryWrapper = new QueryWrapper().eq("problem_id", problemId);
        return selectOneByQuery(queryWrapper).getId();
    }


    /**
     * 根据题目id查询测试样例
     * @param problemId 题目id
     * @return 测试样例
     */
    default String getCases(Long problemId){
        var queryWrapper = new QueryWrapper().eq("problem_id", problemId);
        return selectOneByQuery(queryWrapper).getTestCase();
    }
}
